Skip to content

[Pytorch] Add function to check if EP is bootstrapped - #3321

Merged
phu0ngng merged 1 commit into
NVIDIA:mainfrom
YangFei1990:expose_bootstrap_ep
Aug 5, 2026
Merged

[Pytorch] Add function to check if EP is bootstrapped#3321
phu0ngng merged 1 commit into
NVIDIA:mainfrom
YangFei1990:expose_bootstrap_ep

Conversation

@YangFei1990

@YangFei1990 YangFei1990 commented Aug 5, 2026

Copy link
Copy Markdown
Collaborator

Description

Simple change to expose _BOOTSTRAPPED

Type of change

  • Documentation change (change only to the documentation, either a fix or a new content)
  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • Infra/Build change
  • Code refactoring

Changes

Please list the changes introduced in this PR:

  • Change A
  • Change B

Checklist:

  • I have read and followed the contributing guidelines
  • The functionality is complete
  • I have commented my code, particularly in hard-to-understand areas
  • I have made corresponding changes to the documentation
  • My changes generate no new warnings
  • I have added tests that prove my fix is effective or that my feature works
  • New and existing unit tests pass locally with my changes

@YangFei1990
expose _BOOTSTRAPPED
d032bd4 
Signed-off-by: YangFei1990 <feiw@nvidia.com>
@YangFei1990
YangFei1990 requested a review from ksivaman as a code owner August 5, 2026 22:52
@greptile-apps

greptile-apps Bot commented Aug 5, 2026

Copy link
Copy Markdown
Contributor

Greptile Summary

Adds a public helper for querying whether expert parallelism has been initialized in the current process.

  • Exports is_ep_bootstrapped from the EP module.
  • Returns the existing process-local _BOOTSTRAPPED state without changing initialization or teardown behavior.

Confidence Score: 5/5

The PR appears safe to merge with no identified behavioral, API, or security regressions.

The helper follows the existing EP submodule export convention and accurately reflects the flag updated by successful bootstrap and teardown.

Important Files Changed

Filename Overview
transformer_engine/pytorch/ep.py Adds a correctly exported read-only accessor for the existing EP bootstrap flag; no actionable issues identified.

Reviews (1): Last reviewed commit: "expose _BOOTSTRAPPED" | Re-trigger Greptile

@phu0ngng phu0ngng changed the title Add function to check if EP is bootstrapped [Pytorch] Add function to check if EP is bootstrapped Aug 5, 2026
@phu0ngng
phu0ngng merged commit a7aec21 into NVIDIA:main Aug 5, 2026
10 of 14 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@phu0ngng phu0ngng phu0ngng approved these changes
@ksivaman ksivaman Awaiting requested review from ksivaman ksivaman is a code owner

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@YangFei1990 @phu0ngng